package com.hmdp.utils;

import org.springframework.data.redis.core.StringRedisTemplate;
import org.springframework.stereotype.Component;

import java.time.LocalDateTime;
import java.time.ZoneOffset;
import java.time.format.DateTimeFormatter;

/**
 * Redis ID 工作器，用于生成全局唯一的自增ID。
 */
@Component
public class RedisIdWorker {
    /**
     * 生成下一个唯一 ID
     *
     * @param keyPrefix 键的前缀，用于区分不同类型的ID
     * @return 生成的唯一ID
     */
    public long nextId(String keyPrefix) {
        // 1. 生成时间戳
        LocalDateTime now = LocalDateTime.now(); // 获取当前时间
        long nowSecond = now.toEpochSecond(ZoneOffset.UTC); // 转化为秒级时间戳
        long timestamp = nowSecond - BEGIN_TIMESTAMP; // 减去开始时间，获得相对时间戳

        // 2. 生成序列号
        // 2.1. 获取当前日期，精确到天
        String date = now.format(DateTimeFormatter.ofPattern("yyyy:MM:dd"));
        // 2.2. 调用increment自增，生成当天的自增序列号
        long count = stringRedisTemplate.opsForValue().increment("icr:" + keyPrefix + ":" + date);

        // 3. 将时间戳左移32位，与序列号进行按位或，拼接成唯一的长整型ID
        return timestamp << COUNT_BITS | count;
        // 全局唯一：通过时间戳和自增序列号的组合，保证了生成的 ID 在全局范围内的唯一性。
        // 趋势递增：ID 的前半部分是时间戳，后半部分是自增序列号，保证了生成的 ID 是趋势递增的。
        // 高性能：Redis 的自增操作非常高效，适合高并发场景下的 ID 生成。
    }

    /**
     * 开始时间戳（2022-01-01 00:00:00 UTC）
     */
    private static final long BEGIN_TIMESTAMP = 1640995200L;

    /**
     * 序列号的位数
     */
    private static final int COUNT_BITS = 32;

    private StringRedisTemplate stringRedisTemplate;

    /**
     * 构造函数，注入 StringRedisTemplate
     *
     * @param stringRedisTemplate Redis 操作模板
     */
    public RedisIdWorker(StringRedisTemplate stringRedisTemplate) {
        this.stringRedisTemplate = stringRedisTemplate;
    }
}
